package com.gzh.utils;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import org.junit.jupiter.api.Test;

import java.util.Collections;

public class MyUtils {
    @Test
    public void testCreateCode(){
        FastAutoGenerator.create("jdbc:mysql://127.0.0.1:3306/day530week2?useUnicode=true&characterEncoding=UTF8&useSSL=false&serverTimezone=Asia/Shanghai",
                        "root",
                        "root")
                .globalConfig(builder -> {
                    builder.author("gzh") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .commentDate("yyyy-MM-dd")
                            .outputDir("D:/mytestcode"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.gzh") // 设置父包名
                            //  .controller("actiion")
                            //    .moduleName("system") // 设置父包模块名
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "D:/mytestcode")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("t_stu","t_clazz") // 设置需要生成的表名
                            .addTablePrefix("t_"); // 设置过滤表前缀
                }).strategyConfig(builder -> {
                    builder.entityBuilder()
                            .enableLombok()
                            .columnNaming(NamingStrategy.no_change)
                            .logicDeleteColumnName("del_flag").idType(IdType.AUTO).build();
                    ;
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }

}
